from openai import OpenAI
import time
import os


def get_response(messages):
    client = OpenAI(
        api_key=os.getenv("DASHSCOPE_API_KEY"),
        base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    )
    completion = client.chat.completions.create(
        # model="qwen-turbo-latest",
        model="deepseek-r1-0528",
        messages=messages
    )
    return completion


messages = [{'role': 'system', 'content': '你是一个智慧农业决策者，会给你提供当前室内温度、湿度、风速、光照强度等数据，你要决策风扇是否启动，光照是否调整。'}, {
    'role': 'system', 'content': '数据输出格式：{"fanSpeed": 0, "led": 500}'}, {
    'role': 'system', 'content': '如果输入的格式不对，或者提问无关的内容，你不能回复其他的内容，直接回复：{"fanSpeed": 0, "led": 0}'}]

for i in range(3):
    user_input = input("请输入：")
    start = time.time()
    # 将用户问题信息添加到messages列表中
    messages.append({'role': 'user', 'content': user_input})
    assistant_output = get_response(messages).choices[0].message.content
    # 将大模型的回复信息添加到messages列表中
    messages.append({'role': 'assistant', 'content': assistant_output})

    print(f'模型输出：{assistant_output}')
    print('\n')
    print('耗时：{}秒'.format((time.time() - start)))
